package com.ycwl.aiLive_2d.utils.jwtUtils;


import com.ycwl.aiLive_2d.model.jwt.JwtInfo;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.JwtBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Iterator;
import java.util.Map;

/**
 * @author yangchen
 */
public class JwtAnalysisUtil {

    /**
     * 生成 Token
     *
     * @param jwtInfo
     * @param priKey
     * @param expireTime
     * @return
     * @throws Exception
     */
    public static String generateToken(JwtInfo jwtInfo, byte[] priKey, LocalDateTime expireTime) throws Exception {
        JwtBuilder builder = Jwts.builder().setSubject(jwtInfo.getAccount())
                .claim("userId", jwtInfo.getUserId())
                .claim("roleId", jwtInfo.getRoleId())
                .claim("phone", jwtInfo.getPhone())
                .claim("name", jwtInfo.getName())
                // 返回从1970 00：00：00 到现在的毫秒差（实际的过期时间）
                .claim("expire", expireTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
        return builder.signWith(SignatureAlgorithm.RS256, RsaKeyUtil.getPrivateKey(priKey)).compact();
    }

    /**
     * 解析 Token
     *
     * @param token
     * @param pubKey
     * @return
     * @throws Exception
     */
    public static JwtInfo getInfoFromToken(String token, byte[] pubKey) throws Exception {
        Claims body = (Claims) RsaKeyUtil.parserToken(token, pubKey).getBody();
        Iterator var3 = body.entrySet().iterator();
        while (var3.hasNext()) {
            Map.Entry entry = (Map.Entry) var3.next();
            if (!"sub".equals(entry.getKey())
                    && !"userId".equals(entry.getKey())
                    && !"phone".equals(entry.getKey())
                    && !"roleId".equals(entry.getKey())
                    && !"account".equals(entry.getKey())
                    && !"name".equals(entry.getKey())
                    && !"roleName".equals(entry.getKey())
                    && !"expire".equals(entry.getKey()));
        }

        // convert
        LocalDateTime expireTime = null;
        try {
            // expire time
            Object expire = body.get("expire");
            if (expire != null) {
                expireTime = LocalDateTime.ofInstant(Instant.ofEpochMilli((Long) expire), ZoneId.systemDefault());
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return new JwtInfo(StringUtil.a(body.get("name")),
                StringUtil.a(body.get("userId")),
                StringUtil.a(body.get("roleId")),
                body.getSubject(),
                StringUtil.a(body.get("phone")),
                expireTime);
    }
}
